import input, preprocess, predict

# 主程序
if __name__ == "__main__":
    file_path = "./data/"  # 替换为你的文件路径
    dataframes = input.load_data(file_path)

    # 特征提取
    features = preprocess.preprocess_data(dataframes)

    # 输入需要预测的年份，和主办国家
    year_to_predict = 2028; host_country="United States"; num_of_programs = 350

    # 预测
    predict_data = predict.train_and_predict(features, year_to_predict, host_country, num_of_programs)

    # 评价
    predictions = predict.evaluate_strength(predict_data, features, year_to_predict)

    # 输出结果
    print("\nPredicted Medals for", year_to_predict, "Olympics:")
    print(predictions.to_string(index=False))
    # print(predictions)